import torch
import torchvision

hurtModel=torch.load("/Volumes/study/hurtDetect/healthy-detect/yolov5-master/best.pt")
hurtModel.eval()


example=torch.rand(1,3,640,480)
trace_script_module=torch.jit.trace(hurtModel,example)
trace_script_module.save("/Volumes/study/hurtDetect/healthy-detect/yolov5-master/hurtDetect.pt")